package core

import (
	config2 "gitee.com/cdsailyj/go-vpn/common/cfg"
	"github.com/gin-gonic/gin"
	"gorm.io/driver/mysql"
	"gorm.io/driver/postgres"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"log"
	"os"
	"time"
)

var (
	dbMap = make(map[string]*gorm.DB)
)

func GetDb(c interface{}, uni string, t string) *gorm.DB {
	if dbMap[uni] == nil {
		dbMap[uni] = SetDb(c, t)
	}
	return dbMap[uni]
}

func GetLogger() logger.Interface {
	var loggerDebug logger.Interface
	if gin.Mode() != gin.ReleaseMode {
		loggerDebug = logger.New(
			log.New(os.Stdout, "\r\n", log.LstdFlags),
			logger.Config{
				SlowThreshold: time.Second,
				LogLevel:      logger.Info,
				Colorful:      true,
			},
		)
	}
	return loggerDebug
}

func SetDb(i interface{}, t string) *gorm.DB {
	var db *gorm.DB
	switch t {
	case "mysql":
		db = SetMysqlDb(i.(config2.Mysql))
	case "pgsql":
		db = SetPgsqlDb(i.(config2.Pgsql))
	case "sqlite":
		db = SetSqliteDb(i.(config2.Sqlite))
	}

	return db
}

func SetMysqlDb(c config2.Mysql) *gorm.DB {
	db, err := gorm.Open(mysql.New(mysql.Config{
		DSN:                       c.Dsn,
		DefaultStringSize:         256,   // string 类型字段的默认长度
		DisableDatetimePrecision:  true,  // 禁用 datetime 精度，MySQL 5.6 之前的数据库不支持
		DontSupportRenameIndex:    true,  // 重命名索引时采用删除并新建的方式，MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
		DontSupportRenameColumn:   true,  // 用 `change` 重命名列，MySQL 8 之前的数据库和 MariaDB 不支持重命名列
		SkipInitializeWithVersion: false, // 根据当前 MySQL 版本自动配置
	}), &gorm.Config{Logger: GetLogger()})

	if err != nil {
		log.Println("db screenConnect SetMysqlDb err", err)
		return nil
	}

	// 获取通用数据库对象 sql.DB ，然后使用其提供的功能
	sqlDB, err := db.DB()
	if err != nil {
		log.Println("db screenConnect SetMysqlDb sqlDB err", err)
		return nil
	}
	err = sqlDB.Ping()
	if err != nil {
		log.Println("db screenConnect SetMysqlDb Ping err", err)
		return nil
	}
	// SetMaxIdleConns 用于设置连接池中空闲连接的最大数量。
	sqlDB.SetMaxIdleConns(c.MaxIdleConns)
	// SetMaxOpenConns 设置打开数据库连接的最大数量。
	sqlDB.SetMaxOpenConns(c.MaxOpenConns)
	// SetConnMaxLifetime 设置了连接可复用的最大时间。
	sqlDB.SetConnMaxLifetime(time.Hour)

	return db
}

func SetPgsqlDb(c config2.Pgsql) *gorm.DB {
	db, err := gorm.Open(postgres.New(postgres.Config{DSN: c.Dsn}), &gorm.Config{Logger: GetLogger()})

	if err != nil {
		log.Println("db screenConnect SetPgsqlDb err", err)
		return nil
	}

	// 获取通用数据库对象 sql.DB ，然后使用其提供的功能
	sqlDB, err := db.DB()
	if err != nil {
		log.Println("db screenConnect SetPgsqlDb sqlDB err", err)
		return nil
	}
	err = sqlDB.Ping()
	if err != nil {
		log.Println("db screenConnect SetPgsqlDb Ping err", err)
		return nil
	}
	// SetMaxIdleConns 用于设置连接池中空闲连接的最大数量。
	sqlDB.SetMaxIdleConns(c.MaxIdleConns)
	// SetMaxOpenConns 设置打开数据库连接的最大数量。
	sqlDB.SetMaxOpenConns(c.MaxOpenConns)
	// SetConnMaxLifetime 设置了连接可复用的最大时间。
	sqlDB.SetConnMaxLifetime(time.Hour)

	return db
}

func SetSqliteDb(c config2.Sqlite) *gorm.DB {
	db, err := gorm.Open(sqlite.Open(c.Dsn), &gorm.Config{Logger: GetLogger()})
	if err != nil || db == nil {
		log.Println("db screenConnect SetSqliteDb err", err)
		return nil
	}
	sqlDB, err := db.DB()
	if err != nil {
		log.Println("db screenConnect SetSqliteDb sqlDB err", err)
		return nil
	}
	err = sqlDB.Ping()
	if err != nil {
		log.Println("db screenConnect SetSqliteDb Ping err", err)
		return nil
	}
	return db
}
